The main script for training and logging the experiments is five_fold_train.py

The script that details the model architecture is hnatt.py.